/*
Copyright 2021 Koobaczech

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include QMK_KEYBOARD_H

enum custom_keycodes {
    RGBWINGS = SAFE_RANGE,
};

/* Function to only light up winglets RGB. Must be done in static RGB mode */
bool process_record_user(uint16_t keycode, keyrecord_t *record) {
    switch (keycode) {
    case RGBWINGS:
        if (record->event.pressed) {
            /* Blackout these RGB components */
            rgblight_mode(RGBLIGHT_MODE_STATIC_LIGHT);
            rgblight_setrgb_at(0, 0, 0, 0);
            rgblight_setrgb_at(0, 0, 0, 1);
            rgblight_setrgb_at(0, 0, 0, 4);
            rgblight_setrgb_at(0, 0, 0, 5);
            rgblight_setrgb_at(0, 0, 0, 6);
            rgblight_setrgb_at(0, 0, 0, 7);
            rgblight_setrgb_at(0, 0, 0, 10);
            rgblight_setrgb_at(0, 0, 0, 11);
        }
        break;
    }
    return true;
};

const uint16_t PROGMEM keymaps[][MATRIX_ROWS][MATRIX_COLS] = {
  [0] = LAYOUT_all(
    QK_GESC,  KC_1,     KC_2,     KC_3,  KC_4,  KC_5,  KC_6,  KC_7,    KC_8,  KC_9,     KC_0,     KC_MINS,  KC_EQL,   KC_DEL,    KC_BSPC,
    KC_TAB,   KC_Q,     KC_W,     KC_E,  KC_R,  KC_T,  KC_Y,  KC_U,    KC_I,  KC_O,     KC_P,     KC_LBRC,  KC_RBRC,  KC_BSLS,
    KC_CAPS,  KC_A,     KC_S,     KC_D,  KC_F,  KC_G,  KC_H,  KC_J,    KC_K,  KC_L,     KC_SCLN,  KC_QUOT,  KC_NUHS,  KC_ENT,
    KC_LSFT,  KC_NUBS,  KC_Z,     KC_X,  KC_C,  KC_V,  KC_B,  KC_N,    KC_M,  KC_COMM,  KC_DOT,   KC_SLSH,  KC_RSFT,  MO(1),
    KC_LCTL,  KC_LGUI,  KC_LALT,                              KC_SPC,                   KC_RALT,  KC_RGUI,  KC_APP,   KC_RCTL),

  [1] = LAYOUT_all(
    QK_BOOT,  KC_F1,    KC_F2,    KC_F3,    KC_F4,    KC_F5,    KC_F6,    KC_F7,     KC_F8,    KC_F9,    KC_F10,   KC_F11,   KC_F12,   KC_TRNS,  KC_DEL,
    KC_TRNS,  UG_TOGG,  UG_NEXT,  UG_HUEU,  UG_HUED,  UG_SATU,  UG_SATD,  UG_VALU,   UG_VALD,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,
    KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,   KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,
    KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,   KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_TRNS,  KC_UP,    KC_TRNS,
    KC_TRNS,  KC_TRNS,  KC_TRNS,                                          RGBWINGS,                      KC_TRNS,  KC_LEFT,  KC_DOWN,  KC_RIGHT)
};
